# -*- coding:utf-8 -*-

# @Time    : 2023/5/13 02:24
# @Author  : zengwenjia
# @Email   : zengwenjia@lingxi.ai
# @File    : user_info_extract.py
# @Software: LLM_internal

# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #

from bot.insurance_renren_labels.handle_dialogue_functions import split_content,token_count,merge_result,get_conn,read_data



if __name__ == '__main__':
    data_sql='''SELECT * FROM offline_gpt.result_voice_text a
where not exists (select * from offline_gpt.result_voice_text_result b where a.session_id = b.session_id)'''
    conn=get_conn()
    df=read_data(data_sql,conn)
    print(df.shape)
    df.to_csv("renren_dialogues.csv",index=False)
    print("write success")
    

